package com.example.framework.mybatis.core.handler;

import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.example.framework.mybatis.core.dataobject.BaseDO;
import com.example.framework.web.core.util.WebFrameworkUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;

import java.time.LocalDateTime;
import java.util.Objects;

/**
 * MyBatis的自动填充处理器
 * <p>
 * 写法1；strictInsertFill/strictUpdateFill(metaObject, 实体类属性名称, 实体类属性.class, 填充的数据)
 * <p>
 * 写法2：setFieldValByName(实体类属性名称, 填充的数据, metaObject)
 * <p>
 * 写法3：((T) metaObject.getOriginalObject()).setValue(value)
 */
@Slf4j
public class DefaultDBFieldHandler implements MetaObjectHandler {
    @Override
    public void insertFill(MetaObject metaObject) {
        log.debug("start insert fill");
        String loginUserId = WebFrameworkUtils.getLoginUserId();
        this.strictInsertFill(metaObject, "creator", String.class, loginUserId);
        this.strictInsertFill(metaObject, "updater", String.class, loginUserId);

        if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseDO baseDO) {
            LocalDateTime current = LocalDateTime.now();
            // 创建时间为空，则以当前时间为插入时间
            if (Objects.isNull(baseDO.getCreateTime())) {
                baseDO.setCreateTime(current);
            }
            // 更新时间为空，则以当前时间为更新时间
            if (Objects.isNull(baseDO.getUpdateTime())) {
                baseDO.setUpdateTime(current);
            }
        }
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        log.debug("start update fill ....");

        // 更新时间为空，则以当前时间为更新时间
        Object modifyTime = getFieldValByName("updateTime", metaObject);
        if (Objects.isNull(modifyTime)) {
            setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
        }

        // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
        Object modifier = getFieldValByName("updater", metaObject);
        String userId = WebFrameworkUtils.getLoginUserId();
        if (ObjectUtil.isNotEmpty(userId) && Objects.isNull(modifier)) {
            setFieldValByName("updater", userId, metaObject);
        }
    }
}
